from pathlib import Path

# 项目根目录
ROOT_DIR = Path(__file__).parent.parent
# 数据路径
RAW_DATA_DIR = ROOT_DIR / 'data' / 'raw'
PROCESSED_DATA_DIR = ROOT_DIR / 'data' / 'processed'
# 模型和日志路径
MODELS_DIR = ROOT_DIR / 'models'
LOGS_DIR = ROOT_DIR / 'logs'

# 模型结构参数
DIM_MODEL = 128  # 词向量维度
NUM_HEADS = 4  # 多头自注意力机制
NUM_ENCODER_LAYERS = 2  # encoder层数
NUM_DECODER_LAYERS = 2  # decoder层数

# 训练参数
SEQ_LEN = 30  # 序列长度（输入与输出最大长度）
BATCH_SIZE = 128  # 批次大小
LEARNING_RATE = 1e-3  # 学习率
EPOCHS = 30  # 训练轮数

RANDOM_SEED = 42
